

from ultralytics import YOLO

if __name__ == '__main__':
    # 加载模型
    model = YOLO('models/v8/yolov8.yaml')  # 确保yolov12.yaml路径正确
    
    # 训练模型
    model.train(
        data='data.yaml',  # 检查data.yaml路径及内容是否正确
        epochs=100,
        batch=1,
        workers=20  # 临时将workers设为0排查问题（可选）
    )